package com.atguigu.gmall.common.config;

import com.alibaba.fastjson.JSON;
import com.atguigu.gmall.common.model.GmallCorrelationData;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.concurrent.TimeUnit;

/**
 * @Author atguigu-mqx
 * @Description TODO
 * @Date 2023/9/4 11:16
 * @Version 1.0
 */
@Component
@Slf4j
public class MQProducerAckConfig implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    //  初始化操作
    @PostConstruct
    public void init(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }
    /**
     * 判断当前消息是否到达交换机
     * @param correlationData
     * @param ack
     * @param text
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String text) {
        //  判断
        if (ack){
            //  说明到达交换机
            log.info("说明到达交换机");
        }else {
            log.error("说明没有到达交换机:\t"+text);
            //  调用重试方法.
            this.retrySendMsg(correlationData);
        }
    }



    /**
     * 判断当前消息是否到达队列，如果没有到达对，则这个方法被执行！ 到队列，则这个方法不执行！
     * @param message
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息路由queue失败，应答码={}，原因={}，交换机={}，路由键={}，消息={}",
                replyCode, replyText, exchange, routingKey, message.toString());
        //  消息没有到队列的时候才执行.
        //  只要系统执行到这个方法，就应该执行重复发送
        //  从缓存中获取数据. this.redisTemplate.opsForValue().set(uuid, JSON.toJSONString(gmallCorrelationData),10, TimeUnit.MINUTES);
        //  获取缓存的key  uuid
        String key = message.getMessageProperties().getHeader("spring_returned_message_correlation");
        //  根据key 来获取字符串
        String gmallJson = (String) this.redisTemplate.opsForValue().get(key);
        //  需要将这个字符串转换为CorrelationData
        GmallCorrelationData gmallCorrelationData = JSON.parseObject(gmallJson, GmallCorrelationData.class);
        //  如果是延迟消息就不走重试业务逻辑.
        if (gmallCorrelationData.isDelay()) return;
        //  调用重试方法.
        this.retrySendMsg(gmallCorrelationData);
    }

    private void retrySendMsg(CorrelationData correlationData) {
        //  获取到重试次数
        GmallCorrelationData gmallCorrelationData = (GmallCorrelationData) correlationData;
        int retryCount = gmallCorrelationData.getRetryCount();
        //  判断 0->1->2->3
        if (retryCount>=3){
            System.out.println("次数已到"+retryCount);
            //  如果次数已到，还没有发送出去.
            log.info("次数已到！");
            //  将消息，写入消息记录表。。。。。 做持久化，然后才能人工审核处理！ insert into msg value()
        } else {
            //  发送消息
            System.out.println("次数"+retryCount);
            retryCount++;
            gmallCorrelationData.setRetryCount(retryCount);
            //  将这个最新数据，写回缓存.
            this.redisTemplate.opsForValue().set(gmallCorrelationData.getId(),JSON.toJSONString(gmallCorrelationData),10, TimeUnit.MINUTES);
            //  判断发送消息的类型
            if (gmallCorrelationData.isDelay()){
                //  延迟消息
                this.rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),
                        message -> {
                            //  设置延迟时间
                            message.getMessageProperties().setDelay(1000*gmallCorrelationData.getDelayTime());
                            //  返回
                            return message;
                        },gmallCorrelationData);
                //  不能RabbitService.sendDelayMsg();
            } else {
                //  发送普通消息
                this.rabbitTemplate.convertAndSend(gmallCorrelationData.getExchange(),gmallCorrelationData.getRoutingKey(),gmallCorrelationData.getMessage(),gmallCorrelationData);
                //  不能RabbitService.sendMsg();
            }
       }
    }
}
